the result from torch.concat() is stored in cpu(memory)